/* 
 * GDT ir IDT lentelių aprašai
 */
#include "../include/types.h"
#include "../include/const.h"
#include "../include/mem.h"
#include "../include/descr_tables.h"

/* Kad galima būtų pasiekti funkcijas asemblerio kode */
extern void gdt_flush(u32int);
extern void idt_flush(u32int);
/* Vidinių funkcijų antraštės */
static void init_gdt();
static void gdt_set_gate(s32int,u32int,u32int,u8int,u8int);
static void init_idt();
static void idt_set_gate(u8int,u32int,u16int,u8int);

gdt_entry_t gdt_entries[6]; /* Čia reiks nuspresti kokį skaičių rašyti */
gdt_ptr_t   gdt_ptr;
idt_entry_t idt_entries[256]; /* Tiek turi but */
idt_ptr_t   idt_ptr; 

boolean init_descriptor_tables(){
  short err=FALSE;
  init_gdt();
  init_idt();
  err=TRUE;
  return err;
}

static void init_gdt(){
  gdt_ptr.limit = (sizeof(gdt_entry_t) * 6) - 1; /* Kadangi 6 įeitys bus, tai dauginame iš 6 ir -1 nes nuo 0 skaičiuojame */
  gdt_ptr.base  = (u32int)&gdt_entries; /* Nurodome pradžią lentelės */

  gdt_set_gate(0, 0, 0, 0, 0);                /* Null segmentas */
  gdt_set_gate(1, 0, 0xFFFFFFFF, 0x9A, 0xCF); /* Kodo segmentas */
  gdt_set_gate(2, 0, 0xFFFFFFFF, 0x92, 0xCF); /* Duomenų segmentas */
  gdt_set_gate(3, 0, 0xFFFFFFFF, 0xFA, 0xCF); /* Vartotojo kodo segmentas */
  gdt_set_gate(4, 0, 0xFFFFFFFF, 0xF2, 0xCF); /* Vartotojo duomenų segmentas */
  /* +TSS */
  gdt_flush((u32int)&gdt_ptr); /* Ši funkcija užkrauna GDT lentelę */
}

static void gdt_set_gate(s32int num, u32int base, u32int limit, u8int access, u8int gran){
  gdt_entries[num].base_low    = (base & 0xFFFF); /* Gauname 16 bitų žemiausių  */
  gdt_entries[num].base_middle = (base >> 16) & 0xFF; /* Gauname 8 bitus vidurinius */
  gdt_entries[num].base_high   = (base >> 24) & 0xFF; /* Gauname 8 bitus aukščiausius */

  gdt_entries[num].limit_low   = (limit & 0xFFFF); /* Gauname 16 bitų žemiausių */
  gdt_entries[num].granularity = (limit >> 16) & 0x0F; /* Gauname 16 bitų viresnių limito, kurių 8 bitai žemesni yra prasmingi */

  gdt_entries[num].granularity |= gran & 0xF0; /* Gauname granularity 4 bitus, kurių 2 aukštesnieji yra prasmingi */
  gdt_entries[num].access      = access; /* Išsaugome access 8 bitus */
} 

static void init_idt(){
  idt_ptr.limit = sizeof(idt_entry_t) * 256 -1; /* -1 nes nuo 0 skaičiuojama ir išviso yra 256 įeitys galimos */
  idt_ptr.base  = (u32int)&idt_entries; /* Nurodome pradžią lentelės */

  memset((u8int *)&idt_entries, 0, sizeof(idt_entry_t)*256); /* Visą reikalinga atmintį IDT lentelei inicializuojame nuliais */
  
  /* IRQ lentelės perinstaliavimas */
  outb(0x20, 0x11);
  outb(0xA0, 0x11);
  outb(0x21, 0x20);
  outb(0xA1, 0x28);
  outb(0x21, 0x04);
  outb(0xA1, 0x02);
  outb(0x21, 0x01);
  outb(0xA1, 0x01);
  outb(0x21, 0x0);
  outb(0xA1, 0x0);
  /* Visi skaičiai šešioliktainiai tie patys, nes: visi present, visi ring0 */
  /* Selektorius toks pats visiems */
  idt_set_gate(0, (u32int)isr0, 0x08, 0x8E); 
  idt_set_gate(1, (u32int)isr1, 0x08, 0x8E); 
  idt_set_gate(2, (u32int)isr2, 0x08, 0x8E);
  idt_set_gate(3, (u32int)isr3, 0x08, 0x8E);
  idt_set_gate(4, (u32int)isr4, 0x08, 0x8E);
  idt_set_gate(5, (u32int)isr5, 0x08, 0x8E);
  idt_set_gate(6, (u32int)isr6, 0x08, 0x8E);
  idt_set_gate(7, (u32int)isr7, 0x08, 0x8E);
  idt_set_gate(8, (u32int)isr8, 0x08, 0x8E);
  idt_set_gate(9, (u32int)isr9, 0x08, 0x8E);
  idt_set_gate(10, (u32int)isr10, 0x08, 0x8E);
  idt_set_gate(11, (u32int)isr11, 0x08, 0x8E);
  idt_set_gate(12, (u32int)isr12, 0x08, 0x8E);
  idt_set_gate(13, (u32int)isr13, 0x08, 0x8E);
  idt_set_gate(14, (u32int)isr14, 0x08, 0x8E);
  idt_set_gate(15, (u32int)isr15, 0x08, 0x8E);
  idt_set_gate(16, (u32int)isr16, 0x08, 0x8E);
  idt_set_gate(17, (u32int)isr17, 0x08, 0x8E);
  idt_set_gate(18, (u32int)isr18, 0x08, 0x8E);
  idt_set_gate(19, (u32int)isr19, 0x08, 0x8E);
  idt_set_gate(20, (u32int)isr20, 0x08, 0x8E);
  idt_set_gate(21, (u32int)isr21, 0x08, 0x8E);
  idt_set_gate(22, (u32int)isr22, 0x08, 0x8E);
  idt_set_gate(23, (u32int)isr23, 0x08, 0x8E);
  idt_set_gate(24, (u32int)isr24, 0x08, 0x8E);
  idt_set_gate(25, (u32int)isr25, 0x08, 0x8E);
  idt_set_gate(26, (u32int)isr26, 0x08, 0x8E);
  idt_set_gate(27, (u32int)isr27, 0x08, 0x8E);
  idt_set_gate(28, (u32int)isr28, 0x08, 0x8E);
  idt_set_gate(29, (u32int)isr29, 0x08, 0x8E);
  idt_set_gate(30, (u32int)isr30, 0x08, 0x8E);
  idt_set_gate(31, (u32int)isr31, 0x08, 0x8E);
  idt_set_gate(32, (u32int)irq0, 0x08, 0x8E);
  idt_set_gate(33, (u32int)irq1, 0x08, 0x8E);
  idt_set_gate(34, (u32int)irq2, 0x08, 0x8E);
  idt_set_gate(35, (u32int)irq3, 0x08, 0x8E);
  idt_set_gate(36, (u32int)irq4, 0x08, 0x8E);
  idt_set_gate(37, (u32int)irq5, 0x08, 0x8E);
  idt_set_gate(38, (u32int)irq6, 0x08, 0x8E);
  idt_set_gate(39, (u32int)irq7, 0x08, 0x8E);
  idt_set_gate(40, (u32int)irq8, 0x08, 0x8E);
  idt_set_gate(41, (u32int)irq9, 0x08, 0x8E);
  idt_set_gate(42, (u32int)irq10, 0x08, 0x8E);
  idt_set_gate(43, (u32int)irq11, 0x08, 0x8E);
  idt_set_gate(44, (u32int)irq12, 0x08, 0x8E);
  idt_set_gate(45, (u32int)irq13, 0x08, 0x8E);
  idt_set_gate(46, (u32int)irq14, 0x08, 0x8E);
  idt_set_gate(47, (u32int)irq15, 0x08, 0x8E);
  
  idt_flush((u32int)&idt_ptr); /* Ši funkcija užkrauna IDT lentelę */
}

static void idt_set_gate(u8int num, u32int base, u16int sel, u8int flags){
  idt_entries[num].base_lo = base & 0xFFFF; /* Gauname 16 bitų žemesniųjų */
  idt_entries[num].base_hi = (base >> 16) & 0xFFFF; /* Gauname 16 bitų viresniųjų */

  idt_entries[num].sel     = sel; /* Išsaugome selectorių */
  idt_entries[num].always0 = 0; /* Šitas turi visada būti 0 */
  idt_entries[num].flags   = flags  | 0x60 ; /* Atkomentavos, gausime ring3 privilegijas (user). Dabar yra ring0 */
}
